package cn.itcast.web.shiro;

import cn.itcast.domain.system.Module;
import cn.itcast.domain.system.User;
import cn.itcast.service.system.ModuleService;
import cn.itcast.service.system.UserService;
import org.apache.shiro.authc.*;
import org.apache.shiro.authz.AuthorizationInfo;
import org.apache.shiro.authz.SimpleAuthorizationInfo;
import org.apache.shiro.realm.AuthorizingRealm;
import org.apache.shiro.subject.PrincipalCollection;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.List;

/**
 * 自定义realm
 * 1.作用：是访问我们的权限数据
 * 2.当我们做认证授权操作的时候，需要查询数据库，，在这里完成数据库的查询操作
 *
 */
public class AuthRealm extends AuthorizingRealm {
    /**
     * 认证方法：当调用subject.login()方法的时候，会自动跳转到这里
     * 1.如果当前方法返回null,报错，UnknownAccountException
     * 2.shiro如何进行登陆认证校验？
     * 2.1  用户名是否在数据库存在，我们自己判断，根据用户名查询数据库
     *   A：shiro要知道用户输入的密码
     *   B：shiro要知道数据库中正确的密码
     * @param principals
     * @return
     */
@Autowired
private UserService userService;
@Autowired
private ModuleService moduleService;




    //认证的方法
    //当调用subject.login()方法时，会自动来到这里
    @Override
    protected AuthenticationInfo doGetAuthenticationInfo(AuthenticationToken token) throws AuthenticationException {
        //1.转换
        UsernamePasswordToken upToken = (UsernamePasswordToken) token;
        //获取用户输入的email
        String email = upToken.getUsername();
        //根据用户输入的email去查询数据库

       User user= userService.findByEmail(email);
       if (user==null){
           return null;  //返回null必然报错UnknownAccountException
       }
       //如果不为null就获取数据库中的正确密码密码
        String dbPassword = user.getPassword();


       /*
       参数1.认证的身份对象，这里的参数1通过了subject.getprincipal()获取
       参数2：数据库中正确的密码
       参数3：realm的名称，可以随便给，但是多个realm要保证唯一
        */
        SimpleAuthenticationInfo sai = new SimpleAuthenticationInfo(user, dbPassword, getName());
        return sai;

    }


    //授权方法：当调用subject.checkPermission()方法的时候，自动来到此方法
    @Override
    protected AuthorizationInfo doGetAuthorizationInfo(PrincipalCollection principals) {
        //需求：告诉shiro当前登陆的用户有那些权限
        //1.获取登陆认证的对象，其实就是认证方法返回对象的构造函数的第一个参数
        User user = (User) principals.getPrimaryPrincipal();
        //2.根据用户id,查询用户所具有的所有权限
        List<Module> list = moduleService.findModuleByUserId(user.getId());
        //3.返回，告诉shiro用户的权限
        SimpleAuthorizationInfo sai = new SimpleAuthorizationInfo();
        if (list!=null && list.size()>0){
            for (Module module : list) {
                //添加权限,告诉shiro用户的权限
                sai.addStringPermission(module.getName());
            }
        }
        return sai;
    }
}
